Towards Looped Models Done Right, Part I: Topology, Input Injection, Recurrent-State Design
迈向正确的循环模型:拓扑、输入注入与循环状态设计(第一部分)
🧭 高效循环 LLM 架构设计
Institute of Foundation Models · USC · CMU
系统消融循环LLM的拓扑、输入注入和循环状态设计,证明Huginn架构在稠密和MoE模型上均优于Ouro且参数效率更高
📌 工作参考价值(4 点)
- 循环 Transformer 是低成本部署 LLM 的有效架构:在电商搜索、推荐系统等需线上低延迟、低显存的场景,可采用 prelude–recurrent block–coda sandwich 结构,以相同 FLOPs 将 resident 参数减少约 75%,适合高频更新和边缘部署。
- 把 input injection 看作增强知识检索与上下文利用的专项机制:在做 query 改写、搜索 suggestion、商品描述生成或 RAG-based 推荐时,可专门设计输入注入层,将用户上下文、商品知识等注入循环块,而不用改变整体拓扑。
- 循环状态设计应从简:论文显示 separate H/L recurrent states 和 random initialization 无稳定收益,因此工程实现中不要过度设计循环状态,优先使用简单共享状态和标准初始化,降低实现复杂度。
- MoE 与循环结合可行且路由更均衡:推荐系统的大规模稀疏 MoE 模型可借鉴 Huginn-style loop,用更少 resident 参数维持性能,同时改善专家负载均衡,适合多域/多任务推荐联合训练。